#ifndef HTF_ENGINE_H
#define HTF_ENGINE_H
#ifdef __cplusplus
extern "C" {
#endif

// Adjust to the model's actual input dimension (originally trained with 128)
#define HTF_INPUT_SIZE 128
#define HTF_OUTPUT_SIZE 3

typedef struct {
    float data[HTF_INPUT_SIZE];
} HtfInput;

typedef struct {
    float scores[HTF_OUTPUT_SIZE];
    int predicted_class;
} HtfOutput;

int HtfRunInference(const HtfInput* input, HtfOutput* output);
void HtfEngineInit(void);
void HtfEngineProcess(void);

#ifdef __cplusplus
}
#endif
#endif // HTF_ENGINE_H